Unified Multi-Task Relevance Modeling for E-Commerce: Comparing Task Routing Architectures Across LLMs and Cross-Encoders
电商统一多任务相关度建模:对比LLM与Cross-Encoder的任务路由架构
🧭 多任务相关度建模 · 任务路由架构 ★ 深度解读
Walmart Global Tech
首次对比三种任务路由架构在LLM和Cross-Encoder上的效果,发现解码器仅靠文本前缀时若去掉前缀会崩塌,私有层可完全恢复,多任务训练提升低资源任务最高+14%。
📌 工作参考价值(4 点)
- 多任务框架设计: 将6种实体对关系(query-ad, query-query, ad-ad等)统一用三点相关度分级联合训练,利用数据丰富任务(如query-ad)的知识转移到低资源任务(如product-type-product-type),直接提升低资源任务准确率最高14%,适合电商搜索推荐中多种相关性场景的统一建模。
- 任务路由选择: 对于Cross-Encoder,用文本前缀就够了,去掉前缀换成多分类头几乎无损失(~1%);对于decoder-only LLM,必须加私有Transformer层(MHP),否则去掉文本前缀后模型会崩塌(Gemma-2-2b从87.14%掉到55.59%)。如果用LoRA微调LLM做多任务,务必配套私有层。
- 集成方案: 将LLM和Cross-Encoder的MHP模型做多数投票,利用两类模型的错误多样性(如LLM在抽象任务上更强,Cross-Encoder在词汇丰富任务上更强),在453K测试集上达到89.96%准确率,比文本前缀集成高1.92%,且MHP私有层训练出的模型多样性更强,适合构建高精度离线标注或线上集成。
- 工程效率权衡: Gemma-2-2b(2B)用LoRA(~2M可训参数)搭配MHP后准确率88.04%,接近Llama-3-8b(88.27%),且推理速度快于大模型,适合低延迟线上场景;ModernBERT(396M)用于批量离线标注效果最好(97.81%在ad-ad任务),但泛化到抽象任务弱,可按任务类型选模型。